package org.example.stockdemo.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.example.stockdemo.entity.Stock;
import org.example.stockdemo.entity.StockGroup;
import org.example.stockdemo.entity.StockGroupItem;

import java.util.List;

@Mapper
public interface StockGroupItemMapper extends BaseMapper<StockGroupItem> {
    /**
     * 获取分组中的股票列表
     */
    List<Stock> getStocksByGroupId(@Param("groupId") String groupId);

    /**
     * 获取股票所在的分组列表
     */
    List<StockGroup> getGroupsByStockId(@Param("stockId") String stockId);

    /**
     * 将股票添加到分组
     */
    int addStockToGroup(@Param("stockId") String stockId, @Param("groupId") String groupId);

    /**
     * 从分组中移除股票
     */
    int removeStockFromGroup(@Param("stockId") String stockId, @Param("groupId") String groupId);

    /**
     * 移动股票到另一个分组
     */
    int moveStockToGroup(@Param("stockId") String stockId, @Param("fromGroupId") String fromGroupId, @Param("toGroupId") String toGroupId);
}